package cn.wlh.learn.webmagic.tool;


import cn.hutool.core.date.CalendarUtil;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpUtil;
import cn.hutool.log.Log;
import cn.hutool.log.LogFactory;
import cn.hutool.system.SystemUtil;
import com.aliyun.oss.*;
import com.aliyun.oss.common.comm.ResponseMessage;
import com.aliyun.oss.model.*;
import org.junit.Test;

import java.io.*;
import java.net.URL;
import java.util.Calendar;
import java.util.Date;

/**
 * 阿里oss工具类
 * author zbq 2018-11-05
 * https://blog.csdn.net/z_740771101/article/details/90175069
 */
public class AliOssClientTool {

    private static final  Log logger = LogFactory.get();

    private final String bucketName;//Constant.BUCKET_NAME;
    private final OSS ossclient;

    /**
     * 传入自定义OSS
     * @param bucketName
     * @param ossclient
     */
    public AliOssClientTool(String bucketName, OSS ossclient) {
        this.bucketName = bucketName;
        this.ossclient = ossclient;
    }

    /**
     * 秘钥存在于,系统变量或者环境变量.
     * 注意: 设置环境变量，记得重启idea
     * endpoint   -> aliyun_OSS_endpoint
     * accessKeyId -> aliyun_auth_accessKeyId
     * accessKeySecret -> aliyun_auth_accessKeySecret
     * @param bucketName
     */
    public AliOssClientTool(String bucketName){
        this(bucketName, SystemUtil.get("aliyun_OSS_endpoint"),SystemUtil.get("aliyun_auth_accessKeyId"),SystemUtil.get("aliyun_auth_accessKeySecret"));
    }

    /**
     * 传入秘钥和bucketName
     * @param bucketName
     * @param endpoint
     * @param accessKeyId
     * @param accessKeySecret
     */
    public AliOssClientTool(String bucketName, String endpoint , String accessKeyId , String accessKeySecret ){
        //设置超时机制和重试机制
        ClientBuilderConfiguration conf = new ClientBuilderConfiguration();
        conf.setConnectionTimeout(5000);
        conf.setMaxErrorRetry(3);

        this.ossclient=  new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret,conf);

        this.bucketName = bucketName;

    }


    /**
     * 创建存储空间
     *
     * param ossClient OSS连接
     * param bucketName 存储空间
     * return
     */
    public  String createBucketName(String bucketName) {
        // 存储空间
        final String bucketNames = bucketName;
        if (!ossclient.doesBucketExist(bucketName)) {
            // 创建存储空间
            Bucket bucket = ossclient.createBucket(bucketName);
            logger.info("创建存储空间成功");
            return bucket.getName();
        }
        return bucketNames;
    }
    /**
     * 删除存储空间buckName
     *
     * param ossClient oss对象
     * param bucketName 存储空间
     */
    public  void deleteBucket(String bucketName) {
        ossclient.deleteBucket(bucketName);
        logger.info("删除" + bucketName + "Bucket成功");
    }

    /**
     * 创建模拟文件夹
     *
     * param ossClient oss连接
     * param bucketName 存储空间
     * param folder 模拟文件夹名如"qj_nanjing/"
     * return 文件夹名
     */
    public  String createFolder(String bucketName, String folder) {
        // 文件夹名
        final String keySuffixWithSlash = folder;
        // 判断文件夹是否存在，不存在则创建
        if (!ossclient.doesObjectExist(bucketName, keySuffixWithSlash)) {
            // 创建文件夹
            ossclient.putObject(bucketName, keySuffixWithSlash, new ByteArrayInputStream(new byte[0]));
            logger.info("创建文件夹成功");
            // 得到文件夹名
            OSSObject object = ossclient.getObject(bucketName, keySuffixWithSlash);
            String fileDir = object.getKey();
            return fileDir;
        }
        return keySuffixWithSlash;
    }


    /**
     * 根据key删除OSS服务器上的文件
     *
     * param ossClient oss连接
     * param bucketName 存储空间
     * param folder 模拟文件夹名 如"qj_nanjing/"
     * param key Bucket下的文件的路径名+文件名 如："upload/cake.jpg"
     */
    public  void deleteFile(String bucketName, String folder, String key) {
        ossclient.deleteObject(bucketName, folder + key);
        logger.info("删除" + bucketName + "下的文件" + folder + key + "成功");
    }


    /**
     * 上传图片至OSS
     *
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public  String uploadObjectOSS(String storePath, File file) throws FileNotFoundException {
        return this.uploadObjectOSS(storePath,file.getName(),file,new FileInputStream(file));
    }
    /**
     * 上传图片至OSS
     *
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public  String uploadObjectOSS(String storePath, File file,InputStream is) {
        return this.uploadObjectOSS(storePath,file.getName(),file,is);
    }

    /**
     * 上传图片至OSS
     *
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public  String uploadObjectOSS(String storePath,String fileName, File file,InputStream is) {
        String resultStr = null;
        createFolder(bucketName, storePath);
        try {
            // 文件大小
            Long fileSize = file.length();
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossclient.putObject(bucketName, storePath + fileName, is, metadata);
            // 解析结果
            resultStr = storePath + fileName;
            logger.info("putResult.getETag():"+putResult.getETag());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }


    /**
     * 上传图片至OSS
     *
     * param ossClient oss连接
     * param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * param bucketName 存储空间
     * param storePath 模拟文件夹名 如"qj_nanjing/"
     * return String 返回的唯一MD5数字签名
     */
    public  String uploadObjectOSS(String storePath,String fileName,InputStream is) {
        String resultStr = null;
        createFolder(bucketName, storePath);
        try {

            // 文件大小

            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            // metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
             metadata.setContentDisposition("filename=" + fileName);
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = ossclient.putObject(bucketName, storePath + fileName, is, metadata);
            // 解析结果
            resultStr = storePath + fileName;
            logger.info("putResult.getETag():"+putResult.getETag());
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }



    /**
     * 通过文件名判断并获取OSS服务文件上传时文件的contentType
     *
     * param fileName 文件名
     * return 文件的contentType
     */
    public  String getContentType(String fileName) {
        // 文件的后缀名
        String fileExtension = fileName.substring(fileName.lastIndexOf("."));
        if (".bmp".equalsIgnoreCase(fileExtension)) {
            return "image/bmp";
        }
        if (".gif".equalsIgnoreCase(fileExtension)) {
            return "image/gif";
        }
        if (".jpeg".equalsIgnoreCase(fileExtension) || ".jpg".equalsIgnoreCase(fileExtension)
                || ".png".equalsIgnoreCase(fileExtension)) {
            return "image/jpeg";
        }
        if (".png".equalsIgnoreCase(fileExtension)) {
            return "image/png";
        }
        if (".html".equalsIgnoreCase(fileExtension)) {
            return "text/html";
        }
        if (".txt".equalsIgnoreCase(fileExtension)) {
            return "text/plain";
        }
        if (".vsd".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.visio";
        }
        if (".ppt".equalsIgnoreCase(fileExtension) || "pptx".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.ms-powerpoint";
        }
        if (".doc".equalsIgnoreCase(fileExtension) || "docx".equalsIgnoreCase(fileExtension)) {
            return "application/msword";
        }
        if (".xml".equalsIgnoreCase(fileExtension)) {
            return "text/xml";
        }
        // 默认返回类型
        return "";
    }

    /**
     * 获得可以直接使用的外网域名。
     * 如：
     * https://bucket-name.oss-cn-beijing.aliyuncs.com
     * @return
     */
    public String getExtranetEndpoint(){
        BucketInfo bucketInfo = this.getBucketInfo();
        Bucket bucket = bucketInfo.getBucket();
        // 只是获得 oss-cn-beijing.aliyuncs.com
        String extranetEndpoint = bucket.getExtranetEndpoint();
        // 拼接
        return StrUtil.concat(true,"https://",this.getBucketName(),".",extranetEndpoint);
    }

    /**
     * 获得 用于使用HTTP GET方法访问OSS对象的签名url。
     * 这个一般用于私密的图片，配置秘钥有效期加密。
     * @param path 对应上面的key。
     * @param expirationCalendar 有效时间
     * @return
     */
    public String getPresignedUrl(String path,Calendar expirationCalendar){
        Date expiration = expirationCalendar.getTime();
        URL url = this.getOssclient().generatePresignedUrl(this.getBucketName(), path, expiration);
        return url.toString();
    }

    public BucketInfo getBucketInfo(){
        return this.ossclient.getBucketInfo(this.getBucketName());
    }

    public String getBucketName() {
        return bucketName;
    }

    public OSS getOssclient() {
        return ossclient;
    }

    public static void main(String[] args) throws FileNotFoundException {
        String bucketName = "bucket-name1111";
        AliOssClientTool aliOssClientTool = new AliOssClientTool(bucketName);
        OSS ossClient = aliOssClientTool.getOssclient();
        BucketInfo bucketInfo = ossClient.getBucketInfo(bucketName);
        ResponseMessage response = bucketInfo.getResponse();
        CannedAccessControlList cannedACL = bucketInfo.getCannedACL();

        File file =new File("C:\\Users\\Administrator\\Pictures\\1609328981(1).jpg");
        FileInputStream fis = new FileInputStream(file);

//        String bucketName = createBucketName(AliOssClientTool.bucketName);
        System.out.println(aliOssClientTool.bucketName);
        String storePath = "zbq/ces/";
//        String s = aliOssClientTool.uploadObjectOSS(storePath, file);
//        System.out.println(s);


        // 网络图片上传到oss中。
        byte[] bytes = HttpUtil.downloadBytes("https://t1.huishahe.com/uploads/tu/201810/9999/62b6b2242c.jpg");
//        System.out.println(FileUtil.writeBytes(bytes, "C:\\Users\\Administrator\\Desktop\\aaa.jpg"));

        ByteArrayInputStream byteArrayInputStream = IoUtil.toStream(bytes);

        String s1 = aliOssClientTool.uploadObjectOSS(storePath, "62b6b2242c.jpg", byteArrayInputStream);

        System.out.println(s1);


        // 设置URL过期时间为1小时。

//        CalendarUtil

        OSSObject object = ossClient.getObject(bucketName, s1);


//        Date expiration = new Date(new Date().getTime() +  30 *1000);
        //        URL url = ossClient.generatePresignedUrl(bucketName, s1, expiration);

        Calendar calendar = CalendarUtil.calendar();
        Calendar expiration = CalendarUtil.endOfMinute(calendar);
        String url = aliOssClientTool.getPresignedUrl(s1, expiration);
        System.out.println(url);
    }
}